import os
import shutil

# 文件夹路径
folder_A = r'E:\postgraduate\大论文\img_out\images'  # 多文件的文件夹
folder_B = r'E:\postgraduate\大论文\img_out\YOLOv5识别'  # 少文件的文件夹
folder_C = r'E:\postgraduate\大论文\img_out\242'  # 保存的第三方文件夹

# 如果文件夹C不存在，则创建
if not os.path.exists(folder_C):
    os.makedirs(folder_C)

# 获取文件夹B中的所有文件名（不包含后缀）
files_in_B = {os.path.splitext(file)[0] for file in os.listdir(folder_B)}

# 遍历文件夹A中的文件
for file in os.listdir(folder_A):
    file_name, file_ext = os.path.splitext(file)
    if file_name in files_in_B:
        # 如果文件名（不包含后缀）在B文件夹中存在，则复制到文件夹C
        shutil.copy(os.path.join(folder_A, file), os.path.join(folder_C, file))

print("文件复制完成！")
